import matplotlib.pyplot as plt

# ct
# epochs = [10, 30, 50, 53, 60, 80, 100]
# accuracy = [0.9225, 0.9395, 0.9441, 0.9452, 0.9446, 0.9439, 0.9436]
# f1_scores = [0.9210, 0.9371, 0.9430, 0.9441, 0.9435, 0.9425, 0.9423]
# rt old
# epochs = [10, 30, 50, 70, 78, 90, 100]
# accuracy = [0.9115, 0.9263, 0.9341, 0.9382, 0.9376, 0.9369, 0.9368]
# f1_scores = [0.9100, 0.9240, 0.9330, 0.9371, 0.9366, 0.9358, 0.9357]
# rt new
# epochs = [10, 30, 50, 70, 78, 90, 100]
# accuracy = [0.8815, 0.8963, 0.9041, 0.9082, 0.9066, 0.9049, 0.9040]
# f1_scores = [0.8801, 0.8940, 0.9030, 0.9071, 0.9055, 0.9041, 0.9033]

# sc new
# epochs = [10, 20, 30, 40, 50, 60, 70, 78, 90, 100]
# accuracy = [0.2193, 0.4831, 0.6281, 0.7815, 0.8463, 0.9041, 0.9452, 0.9446, 0.9439, 0.9436]
# f1_scores = [0.2181, 0.4823, 0.6270, 0.7801, 0.8440, 0.9030, 0.9441, 0.9435, 0.9425, 0.9423]
#
# epochs = [40, 50, 60, 70, 78, 90, 100]
# accuracy = [0.7815, 0.8663, 0.9241, 0.9452, 0.9452, 0.9439, 0.9436]
# f1_scores = [0.7701, 0.8540, 0.9130, 0.9341, 0.9340, 0.9325, 0.9323]


epochs = [10, 20, 30, 40, 50, 60, 70, 82, 90, 100]
accuracy = [0.4831, 0.6081, 0.6915, 0.7773, 0.8501, 0.8982, 0.9066, 0.9079, 0.9040, 0.9039]
f1_scores = [0.4823, 0.6070, 0.6901, 0.7750, 0.8500, 0.8971, 0.9055, 0.9061, 0.9033, 0.9032]


# 绘制折线
plt.plot(epochs, accuracy, label='Accuracy', marker='o')
plt.plot(epochs, f1_scores, label='F1 Score', marker='d')

# 添加x轴标签
plt.xlabel('Epoch')
# 添加y轴标签
plt.ylabel('')
# 添加图表标题
plt.title('')
# 添加图例
plt.legend()

# 获取当前坐标轴对象
ax = plt.gca()
# 设置顶部和右侧边框颜色为透明
ax.spines['top'].set_color('none')
ax.spines['right'].set_color('none')

# 显示图表
plt.show()